"""
运行英语教学Agent的脚本
"""
import os
import sys
from dotenv import load_dotenv
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from english_tutor_agent.agent import root_agent

def main():
    """启动英语教学Agent"""
    # 加载环境变量
    load_dotenv("english_tutor_agent/.env")
    
    # 检查API密钥是否已设置
    if os.environ.get("GOOGLE_API_KEY") == "YOUR_API_KEY_HERE" and not os.environ.get("GOOGLE_GENAI_USE_VERTEXAI") == "TRUE":
        print("错误: 请在 english_tutor_agent/.env 文件中设置你的 GOOGLE_API_KEY")
        sys.exit(1)
    
    # 创建会话服务
    session_service = InMemorySessionService()
    
    # 创建Runner
    runner = Runner(
        agent=root_agent,
        app_name="english_tutor_app",
        session_service=session_service
    )
    
    print("启动英语教学Agent...")
    print("使用 'adk web' 命令在浏览器中与Agent交互")
    print("或者使用 'adk run english_tutor_agent' 命令在终端中与Agent交互")
    
    # 这里我们不直接运行Agent，而是提示用户使用ADK的命令行工具

if __name__ == "__main__":
    main()